package com.cn.demo4.util;

import org.apache.commons.fileupload.FileItem;
import org.apache.commons.fileupload.FileUploadException;
import org.apache.commons.fileupload.disk.DiskFileItemFactory;
import org.apache.commons.fileupload.servlet.ServletFileUpload;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpSession;
import java.io.*;
import java.util.Date;
import java.util.Iterator;
import java.util.List;

public class LoadImg {
    /**
     * 将页面传递过来的所有数据保存到request里面
     *
     * @param req
     */
    public static void loadImg(HttpServletRequest req) {


//        判断当前传输过来的格式是否为formData格式
        if (ServletFileUpload.isMultipartContent(req)) {

            try {
                // 1. 创建DiskFileItemFactory对象，设置缓冲区大小和临时文件目录
                DiskFileItemFactory factory = new DiskFileItemFactory();
                // System.out.println(System.getProperty("java.io.tmpdir"));//默认临时文件夹

                // 2. 创建ServletFileUpload对象，并设置上传文件的大小限制。
                ServletFileUpload sfu = new ServletFileUpload(factory);
                sfu.setSizeMax(1024 * 1024);// 以byte为单位 不能超过10M 1024byte =
                // 1kb 1024kb=1M 1024M = 1G
                sfu.setHeaderEncoding("utf-8");
                // 3.
                // 调用ServletFileUpload.parseRequest方法解析request对象，得到一个保存了所有上传内容的List对象。
                @SuppressWarnings("unchecked")
                List<FileItem> fileItemList = sfu.parseRequest(req);
                Iterator<FileItem> fileItems = fileItemList.iterator();

                // 4. 遍历list，每迭代一个FileItem对象，调用其isFormField方法判断是否是上传文件
                while (fileItems.hasNext()) {
                    FileItem fileItem = fileItems.next();
                    // 普通表单元素
                    if (fileItem.isFormField()) {
                        String name = fileItem.getFieldName();// name属性值
                        String value = fileItem.getString("utf-8");// name对应的value值
                        req.setAttribute(name, value);
                        System.out.println(name + " = " + value);
                    }
                    // <input type="file">的上传文件的元素
                    else {
                        String fileName = fileItem.getName();// 文件名称a.jpg
                        String name = fileItem.getFieldName();// 前台ajax字段名（key值）
                        //System.out.println("原文件名：" + fileName);// Koala.jpg

//                        String suffix = fileName.substring(fileName.lastIndexOf('.'));
                        //System.out.println("扩展名：" + suffix);// .jpg

                        // 新文件名（唯一）
                        String newFileName = new Date().getTime() + "----" + fileName;
//                        1607324741871--a.jpg
                        //System.out.println("新文件名：" + newFileName);// image\1478509873038.jpg
                        String user_img = MyPath.LOAD_PATH.getPath() + newFileName;//插入数据库url


                        // 5. 调用FileItem的write()方法，写入文件
//                        File file = new File(MyPath.IMG_PATH.getPath() + user_img);//路径改成自己服务器存放图片的路径，如果是本机的直接写磁盘路径，远程服务器的话最好研究一下ftp的协议，这个方法一样，就是要多一个类进行远程接口调用

                        req.setAttribute(name, user_img);

//                        根据当前的路径去查询是否有当前文件夹，如果没有，则创建
                        File file = IOTool.createImgPath(newFileName);
//                        System.out.println(file.getAbsolutePath());
                        fileItem.write(file);
                        createFile(file.getPath(), newFileName);
                        // 6. 调用FileItem的delete()方法，删除临时文件
                        fileItem.delete();


                    }
                }
            } catch (FileUploadException e) {
                e.printStackTrace();
            } catch (Exception e) {
                e.printStackTrace();
            }

        }
    }


    /**
     * 将传递过来的路径复制到out里面的文件夹
     *
     * @param path     路径，整個webxi下面的loadImg的路径加上图片名称
     * @param fileName out下面的loadImg加上图片的名称
     */
    private static void createFile(String path, String fileName) {
        File file = new File(path);
        try {
            InputStream inputStream = new FileInputStream(file);
            int result = 0;
            String outPath = "D:\\IdeaProjects\\demo4\\out\\artifacts\\demo4_war_exploded\\";
            outPath += MyPath.LOAD_PATH.getPath() + fileName;
            OutputStream outputStream = new FileOutputStream(outPath);
            while ((result = inputStream.read()) != -1) {
                outputStream.write(result);
            }
        } catch (FileNotFoundException e) {
            e.printStackTrace();
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
}